from dataset.imagenet import build_imagenet, build_dar

def build_dataset(args, **kwargs):
    # images
    if args.dataset == 'imagenet':
        return build_imagenet(args, **kwargs)
    if args.dataset == 'dar':
        return build_dar(args, **kwargs)
    
    raise ValueError(f'dataset {args.dataset} is not supported')